# Date： 2023/2/24 20:37

filename='student.txt'

# 定义一个主函数
def main():
    while True:
        menum()
        try:
            choice = int(input('请选择'))
            if choice in [0, 1, 2, 3, 4, 5, 6, 7]:
                if choice == 0:
                    answer = input('您确定要退出系统吗？y/n')
                    if answer == 'y' or answer == 'Y':
                        print('谢谢您的使用！！！')
                        break  # 退出系统
                    else:
                        continue
                elif choice == 1:
                    insert()  # 录入学生信息
                elif choice == 2:
                    search()  # 查找学生信息
                elif choice == 3:
                    delete()  # 删除学生信息
                elif choice == 4:
                    update()  # 修改学生信息
                elif choice == 5:
                    sort()  # 排序
                elif choice == 6:
                    total()  # 统计学生总人数
                elif choice == 7:
                    showAllMsg()  # 显示所有学生信息
        except:
                print('请选择正确的菜单选项！')
                continue


# 主菜单
def menum():
    print('============================ 学生信息管理系统 ===============================================')
    print('-------------------------------- 功能菜单 --------------------------------------------------')
    print('\t\t\t\t\t 1. 录入学生信息')
    print('\t\t\t\t\t 2. 查找学生信息')
    print('\t\t\t\t\t 3. 删除学生信息')
    print('\t\t\t\t\t 4. 修改学生信息')
    print('\t\t\t\t\t 5. 排序')
    print('\t\t\t\t\t 6. 统计学生总人数')
    print('\t\t\t\t\t 7. 显示所有学生信息')
    print('\t\t\t\t\t 0. 退出')
    print('******************************************************************************************')

# 1. 录入学生信息
def insert():
    student_list=[]
    while True:
        id=input('请输入学生ID 例如：1001: ')
        if not id:
            break
        name=input('请输入学生姓名: ')
        if not name:
            break
        try:
            english=int(input('请输入英语成绩：'))
            python=int(input('请输入 python 成绩：'))
            java=int(input('请输入 java 成绩：'))
        except:
            print('输入无效，不是整数类型，请重新输入!!!')
            continue
        # 将录入的学生成绩保存到字典
        student={'id': id, 'name': name, 'english':english,'python':python, 'java':java}
        # 将学生的信息添加到列表中
        student_list.append(student)
        answer=input('是否继续添加 yes/no ?')
        if answer=='yes':
            continue
        else:
            break

    # 调用 save() 函数
    save(student_list)
    print('学生信息录入完毕！')


def save(lst):
    try:
        stu_txt=open(filename, 'a', encoding='utf-8')
    except:
        stu_txt=open(filename, 'w', encoding='utf-8')
    for item in lst:
        stu_txt.write(str(item) + '\n')
    stu_txt.close()


# 2. 查找学生信息
def search():
    pass


# 3. 删除学生信息
def delete():
    pass


# 4. 修改学生信息
def update():
    pass


# 5. 排序
def sort():
    pass


# 6. 统计学生总人数
def total():
    pass


# 7. 显示所有学生信息
def showAllMsg():
    pass


if __name__ == '__main__':
    main()
